package com.swallow.auth.common.utils;

import cn.hutool.core.codec.Base64Decoder;
import cn.hutool.core.codec.Base64Encoder;
import org.apache.commons.lang3.StringUtils;

import javax.crypto.Cipher;
import javax.crypto.spec.IvParameterSpec;
import javax.crypto.spec.SecretKeySpec;
import java.nio.charset.StandardCharsets;
import java.util.Objects;
import java.util.Random;

/**
 * @author: yangjie.deng@resico.cn
 * @since: 2024-04-26 10:09:59
 * @version: v1.0.0
 * @describe: AES 加解密
 */
public class AESUtils {
    /** 加密模式之 CBC，算法/模式/补码方式 */
    private static final String AES_CBC = "AES/CBC/PKCS5Padding";

    /** 加密模式之 CFB，算法/模式/补码方式 */
    private static final String AES_CFB = "AES/CFB/PKCS5Padding";

    /** AES 中的 IV 必须是 16 字节（128位）长 */
    private static final Integer IV_LENGTH = 16;

    /***
     * <h2>String 转 byte</h2>
     * @param str 需要转换的字符串
     */
    public static byte[] getBytes(String str){
        return StringUtils.isBlank(str) ? null : str.getBytes(StandardCharsets.UTF_8);
    }

    /***
     * <h2>初始化向量（IV），它是一个随机生成的字节数组，用于增加加密和解密的安全性</h2>
     */
    public static String getIV(){
        String str = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789";
        Random random = new Random();
        StringBuilder sb = new StringBuilder();
        for(int i = 0 ; i < IV_LENGTH ; i++){
            int number = random.nextInt(str.length());
            sb.append(str.charAt(number));
        }
        return sb.toString();
    }

    /***
     * <h2>获取一个 AES 密钥规范</h2>
     */
    public static SecretKeySpec getSecretKeySpec(String key){
        return new SecretKeySpec(Objects.requireNonNull(getBytes(key)), "AES");
    }

    public static void main(String[] args) {
/*        String text = "Zlb@123456";
        String key  = "AES@zlb@prod@key"; // 16字节的密钥
        String iv  = getIV();
        String iv2  = getIV();

        String encryptTextCBC = AESCBCEncrypt(text, key, iv);
        System.out.println("CBC 加密IV：" + iv);
        System.out.println("CBC 加密后内容：" + encryptTextCBC);
        System.out.println("CBC 解密后内容：" + decrypt(encryptTextCBC, key, iv2, AES_CBC));
        System.out.println();*/

        ///     iv: 9d487f08e3674ae2
        //    key: 051487b9f9b666e1
//        password:
//        encrypt:
//        iv: 9d487f08e3674ae2
//        key: 051487b9f9b666e1
        String encrypt = encrypt("Zlb@123456", "051487b9f9b666e1", "9d487f08e3674ae2", AES_CBC);

        System.out.println(encrypt);
    }

    /**
     *  采用CFB 模式加密
     *
     * @param text 加密内容
     * @param key  加密密钥
     * @param iv   初始化向量
     * @return     加密结果
     */
    public static String AESCFBEncrypt(String text, String key, String iv) {
        return encrypt(text, key, iv, AES_CFB);
    }

    /**
     *  采用CFB模式解密
     *
     * @param text 解密内容
     * @param key  解密密钥
     * @param iv   初始化向量
     * @return     加密结果
     */
    public static String AESCFBDecrypt(String text, String key, String iv) {
        return encrypt(text, key, iv, AES_CFB);
    }

    /**
     *  采用CBC 模式进行加密
     *
     * @param text 加密内容
     * @param key  加密密钥
     * @param iv   初始化向量
     * @return     加密结果
     */
    public static String AESCBCEncrypt(String text, String key, String iv) {
        return encrypt(text, key, iv, AES_CBC);
    }

    /**
     *  采用CBC 模式进行解密
     *
     * @param text 解密内容
     * @param key  解密密钥
     * @param iv   初始化向量
     * @return     加密结果
     */
    public static String AESCBCDecrypt(String text, String key, String iv) {
        return decrypt(text, key, iv, AES_CBC);
    }


    /**
     * <h2>加密 - 自定义加密模式</h2>
     * @param text 需要加密的文本内容
     * @param key 加密的密钥 key
     * @param iv 初始化向量
     * @param mode 加密模式
     * */
    private static String encrypt(String text, String key, String iv, String mode){
        if (StringUtils.isAnyBlank(text, key, iv)) {
            return null;
        }

        try {
            // 创建AES加密器
            Cipher cipher = Cipher.getInstance(mode);

            SecretKeySpec secretKeySpec = getSecretKeySpec(key);

            cipher.init(Cipher.ENCRYPT_MODE, secretKeySpec, new IvParameterSpec(Objects.requireNonNull(getBytes(iv))));

            // 加密字节数组
            byte[] encryptedBytes = cipher.doFinal(Objects.requireNonNull(getBytes(text)));

            // 将密文转换为 Base64 编码字符串
            return Base64Encoder.encode(encryptedBytes);
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
    }

    /**
     * <h2>解密 - 自定义加密模式</h2>
     * @param text 需要解密的文本内容
     * @param key 解密的密钥 key
     * @param iv 初始化向量
     * @param mode 加密模式
     * */
    private static String decrypt(String text, String key, String iv, String mode){
        if (StringUtils.isAnyBlank(text, key, iv)) {
            return null;
        }
        // 将密文转换为16字节的字节数组
        byte[] textBytes = Base64Decoder.decode(text);

        try {
            // 创建AES加密器
            Cipher cipher = Cipher.getInstance(mode);

            SecretKeySpec secretKeySpec = getSecretKeySpec(key);

            cipher.init(Cipher.DECRYPT_MODE, secretKeySpec, new IvParameterSpec(Objects.requireNonNull(getBytes(iv))));

            // 解密字节数组
            byte[] decryptedBytes = cipher.doFinal(textBytes);

            // 将明文转换为字符串
            return new String(decryptedBytes, StandardCharsets.UTF_8);
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
    }
}
